package com.j.lemon.learn.clickhouse;

import com.alibaba.druid.pool.DruidDataSource;
import com.j.lemon.learn.utils.SequenceUtil;
import com.opencsv.CSVReader;
import org.apache.commons.lang3.RandomStringUtils;
import org.springframework.jdbc.core.JdbcTemplate;
import ru.yandex.clickhouse.ClickHouseArray;

import java.io.FileReader;
import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.time.Instant;
import java.time.LocalDateTime;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * @author lijunjun
 */
public class CkTest2 {
    public static void main(String[] args) throws SQLException {
        final ExecutorService executorService = new ThreadPoolExecutor(80, 80, 0, TimeUnit.MILLISECONDS, new ArrayBlockingQueue<>(10000), r -> new Thread(r, "executeSql-thread"));
        String driverClassName = "ru.yandex.clickhouse.ClickHouseDriver";
        String url = "jdbc:clickhouse://10.28.149.247:8123/default";
        String userName = "audit4a";
        String password = "rryzggTR";
        DruidDataSource dataSource = new DruidDataSource();
        dataSource.setDriverClassName(driverClassName);
        dataSource.setUrl(url);
        dataSource.setUsername(userName);
        dataSource.setPassword(password);
        dataSource.setTestWhileIdle(true);
        dataSource.setMaxActive(100);
        dataSource.setMaxWait(60000);
        dataSource.setTestOnBorrow(false);


        Connection connection = dataSource.getConnection();
        connection.setAutoCommit(false);
        String sql = "insert into behavior_log_id_test values (\n" +
                "?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,\n" +
                "?,?,?,?,?,?)";
        PreparedStatement ps = connection.prepareStatement(sql);
        String csvFile = "/Users/lijunjun/Desktop/learn/alibabaUserBehaviorDatasets/split1_behavior_log/behavior_log_test_" + 1 + ".csv";
        CSVReader reader = null;
        try {
            reader = new CSVReader(new FileReader(csvFile));
            String[] line;
            List<String[]> data = new ArrayList<>();
            line = reader.readNext();
            System.out.println(Arrays.toString(line));
            data.add(line);
            List<String[]> aaa = new ArrayList<>(data);
            data.clear();
            for (String[] datum : aaa) {
                ps.setLong(1, SequenceUtil.genLongSeqId());
                ps.setInt(2, Integer.parseInt(datum[0]));
                ps.setString(3, datum[1]);
                ps.setString(4, datum[2]);
                ps.setString(5, datum[3]);
                LocalDateTime localDateTime = LocalDateTime.ofInstant(Instant.ofEpochMilli(Long.parseLong(datum[4])), ZoneOffset.of("+8"));
                String format = localDateTime.format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
                ps.setString(6, format);
                addExtColumns(ps);
                ps.addBatch();
            }
            long l = System.currentTimeMillis();
            ps.executeBatch();
            connection.commit();
            System.out.println("1 records batch inserted cost:" + (System.currentTimeMillis() - l));
        } catch (IOException e) {
            e.printStackTrace();
        }
//        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
//        Map<String, Object> map = jdbcTemplate.queryForMap("select groupArray(userid) ,btag,count() from behavior_log_all where btag = 'pv'  group by btag");
//        ClickHouseArray o = (ClickHouseArray) map.get("groupArray(userid)");
//        long [] array = (long[]) o.getArray();
//        System.out.println(array.length);


    }

    private static void addExtColumns(PreparedStatement ps) throws SQLException {
        for (int i = 1; i < 501; i++) {
            ps.setString(i + 6, RandomStringUtils.randomAlphanumeric(10));
        }
    }
}
